import erniebot


# 初始化 Erniebot
def initialize_erniebot():
    erniebot.api_type = 'aistudio'
    erniebot.access_token = '721efe09951433260c4ab297c2830cffc82e2d77'

def translate(user_query,type_trsnslate="英语"):
    initialize_erniebot()

    question = f"""
    我希望你能充当{type_trsnslate}翻译、拼写纠正者和改进者。并且只输出翻译后的句子。
    我将用任何语言与你交谈，你将检测语言，翻译它，并在我的文本的更正和改进版本中用{type_trsnslate}回答。
    我希望你用更漂亮、更优雅、更高级的{type_trsnslate}单词和句子来取代我的简化A0级单词和句子。
    保持意思不变，但让它们更有文学性。我希望你只回答更正，改进，而不是其他，不要写解释。
    
    我需要翻译的话是： {user_query}
    """

    response = erniebot.ChatCompletion.create(
        model='ernie-3.5',
        messages=[
            {'role': 'user', 'content': question},
        ],
        temperature=0.8
    )

    return response.get_result()


# 调用方式示例（在其他文件中，首先要导入这个函数）
# user_question = input()
# result = translate(user_question)
# print(result)
